﻿def ReplaceHeavyWeaponTeam():
	RemoveWeaponUpgrades()
	return True

def RemoveWeaponUpgrades():
	e = List.SelectedEntry
	for upgrade in e.Upgrades:
		if upgrade.Equipment.Category.endswith("Weapons"):
			upgrade.Applied = False
	
def RemoveWeapons():
	e = List.SelectedEntry
	for unit in e.Units:
		for eq in unit.Equipment:
			if eq.Equipment.Category.endswith("Weapons"):
				eq.Equipped = False

def AddSpecialWeapon(m):
	e = List.SelectedEntry
	max = int(m)
	count = 0

